package com.urea.system.service.impl;

import cn.hutool.core.io.FileUtil;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.urea.common.exception.ServiceException;
import com.urea.common.utils.file.FileUtils;
import com.urea.system.domain.FileInfo;
import com.urea.system.mapper.FileInfoMapper;
import com.urea.system.service.IFileInfoService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import javax.servlet.http.HttpServletResponse;
import javax.sql.rowset.serial.SerialException;
import java.io.File;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.util.List;
import java.util.Objects;

/**
 * 文件Service业务层处理
 * 
 * @author liuzd
 * @date 2023-08-01
 */
@Service
public class FileInfoServiceImpl extends ServiceImpl<FileInfoMapper, FileInfo> implements IFileInfoService
{
    @Autowired
    private FileInfoMapper fileInfoMapper;

    /**
     * 查询文件
     * 
     * @param id 文件主键
     * @return 文件
     */
    @Override
    public FileInfo selectFileInfoById(Long id)
    {
        return fileInfoMapper.selectFileInfoById(id);
    }

    /**
     * 查询文件列表
     * 
     * @param FileInfo 文件
     * @return 文件
     */
    @Override
    public List<FileInfo> selectFileInfoList(FileInfo FileInfo)
    {
        return fileInfoMapper.selectFileInfoList(FileInfo);
    }

    /**
     * 新增文件
     * 
     * @param FileInfo 文件
     * @return 结果
     */
    @Override
    public int insertFileInfo(FileInfo FileInfo)
    {
        return fileInfoMapper.insertFileInfo(FileInfo);
    }

    /**
     * 修改文件
     * 
     * @param FileInfo 文件
     * @return 结果
     */
    @Override
    public int updateFileInfo(FileInfo FileInfo)
    {
        return fileInfoMapper.updateFileInfo(FileInfo);
    }

    /**
     * 批量删除文件
     * 
     * @param ids 需要删除的文件主键
     * @return 结果
     */
    @Override
    public int deleteFileInfoByIds(Long[] ids)
    {
        return fileInfoMapper.deleteFileInfoByIds(ids);
    }

    /**
     * 删除文件信息
     * 
     * @param id 文件主键
     * @return 结果
     */
    @Override
    public int deleteFileInfoById(Long id)
    {
        return fileInfoMapper.deleteFileInfoById(id);
    }

    /**
     * 文件下载
     *
     * @param response 响应对象
     * @param id       文件id
     */
    @Override
    public void downLoad(HttpServletResponse response, Long id) {
        // 获取文件
        FileInfo fileInfo = this.getById(id);
        if (Objects.isNull(fileInfo)){
            throw new ServiceException("文件不存在!");
        }
        File file = new File(fileInfo.getPath());
        try {
            FileUtils.setAttachmentResponseHeader(response,fileInfo.getName());
            FileUtil.writeToStream(file,response.getOutputStream());
        } catch (UnsupportedEncodingException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}
